package array.removeelement;

import java.util.Arrays;

/**
 * @ClassName SortedSquares
 * @Description TODO
 * @Author lenovo
 * @Date 2023-02-10 17:18
 * @Version 1.0
 * @Comment Magic. Do not touch.
 * If this comment is removed. the program will blow up
 */
public class SortedSquares {

    /**
     * 977. 有序数组的平方
     * 给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。
     * <p>
     * 示例 1：
     * <p>
     * 输入：nums = [-4,-1,0,3,10]
     * 输出：[0,1,9,16,100]
     * 解释：平方后，数组变为 [16,1,0,9,100]
     * 排序后，数组变为 [0,1,9,16,100]
     * 示例 2：
     * <p>
     * 输入：nums = [-7,-3,2,3,11]
     * 输出：[4,9,9,49,121]
     *
     * @param nums
     * @return
     */
    public static int[] sortedSquares(int[] nums) {

        for (int i = 0; i < nums.length; i++) {
            int multiply = nums[i] * nums[i];
            nums[i] = multiply;
        }

        Arrays.sort(nums);
        return nums;
    }


    /**
     * 双指针法
     *
     * @param nums
     * @return
     */
    public static int[] sortedSquares1(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];
        for (int i = 0, j = n - 1, pos = n - 1; i <= j; ) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                ans[pos] = nums[i] * nums[i];
                ++i;
            } else {
                ans[pos] = nums[j] * nums[j];
                --j;
            }
            --pos;
        }
        return ans;
    }


    static class Solution {

        /**
         * 有序数组的平方--双指针法
         *
         * @param nums
         * @return
         */
        public int[] sortedSquares(int[] nums) {

            int left = 0;
            int right = nums.length - 1;
            int index = nums.length - 1;
            int[] result = new int[nums.length];
            while (left <= right) {

                if (nums[left] * nums[left] > nums[right] * nums[right]) {
                    result[index--] = nums[left] * nums[left];
                    left++;
                } else {
                    result[index--] = nums[right] * nums[right];
                    right--;
                }
            }
            return result;
        }
    }


    public static void main(String args[]) {
        int[] nums = new int[]{-5, -1, 2, 3, 4};
        int[] numsa = new int[]{-4, -1, 0, 3, 10};
        Solution solution = new Solution();
        int[] ints = solution.sortedSquares(numsa);
        for (int i = 0; i < ints.length; i++) {
            System.out.println(ints[i]);
        }
    }

}